import torch

DEVICE = torch.device("cuda" if torch.cuda.is_available() and False else "cpu")
lr = 1e-4
class_num = 5
checkpoint_folder = "checkpoints"
